python - numpy array set ones between two values, fast